package cn.iocoder.yudao.module.hs.framework.datapermission.config;

import cn.iocoder.yudao.framework.datapermission.core.rule.dept.DeptDataPermissionRuleCustomizer;
import cn.iocoder.yudao.module.hs.dal.dataobject.entryorder.EntryOrderDO;
import cn.iocoder.yudao.module.hs.dal.dataobject.material.MaterialDO;
import cn.iocoder.yudao.module.hs.dal.dataobject.materialbox.MaterialBoxDO;
import cn.iocoder.yudao.module.hs.dal.dataobject.picktask.PickTaskDO;
import cn.iocoder.yudao.module.hs.dal.dataobject.qchistory.QcHistoryDO;
import cn.iocoder.yudao.module.hs.dal.dataobject.qcorder.QcOrderDO;
import cn.iocoder.yudao.module.hs.dal.dataobject.receiveorder.ReceiveOrderDO;
import cn.iocoder.yudao.module.hs.dal.dataobject.rfid.RfidDO;
import cn.iocoder.yudao.module.hs.dal.dataobject.shiporder.ShipOrderDO;
import cn.iocoder.yudao.module.hs.dal.dataobject.tray.TrayDO;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * hs 模块的数据权限 Configuration
 *
 * @author 航盛
 */
@Configuration(proxyBeanMethods = false)
public class HSDataPermissionConfiguration {

    @Bean
    public DeptDataPermissionRuleCustomizer hsDeptDataPermissionRuleCustomizer() {
        return rule -> {
            rule.addDeptColumn(MaterialDO.class);
            rule.addDeptColumn(MaterialBoxDO.class);
            rule.addDeptColumn(TrayDO.class);
            rule.addDeptColumn(RfidDO.class);

            rule.addDeptColumn(ShipOrderDO.class);
            rule.addDeptColumn(EntryOrderDO.class);
            rule.addDeptColumn(ReceiveOrderDO.class);
            rule.addDeptColumn(PickTaskDO.class);
            rule.addDeptColumn(QcOrderDO.class);
            rule.addDeptColumn(QcHistoryDO.class);
        };
    }

}
